#将低分辨率图转为高分辨率
#将图片转为高清的图片并存储到high_resolution 文件夹中

import cv2
from cv2 import dnn_superres
import os
import sys

# 存储位置
output_dir = 'high_resolution'
origin_dir = 'low_resolution'

if not os.path.exists(output_dir):
    os.makedirs(output_dir)

for filenames in os.listdir(origin_dir):
    print(filenames)
    sr = dnn_superres.DnnSuperResImpl_create()
    img = cv2.imread(origin_dir + '/' +filenames)

    path = "models/EDSR_x4.pb" 
    sr.readModel(path) 
    sr.setModel("edsr", 4) # 通过传递倍率值和上采样率来设置模型
    result = sr.upsample(img) #放大输入图像
    cv2.imwrite(output_dir + "/" + str(filenames), result)#保存输出结果到输出文件夹
